package com.junsi.framework.datasource;

import java.util.Map;
import javax.sql.DataSource;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

import com.junsi.common.config.datasource.DynamicDataSourceContextHolder;

/**
 * 动态数据源
 * 
 * @author junsi
 */
public class DynamicDataSource extends AbstractRoutingDataSource
{
    public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources)
    {
        // 设置默认 DataSource，最终给 resolvedDefaultDataSource 初始化对象
        super.setDefaultTargetDataSource(defaultTargetDataSource);
        // 配置多个数据源
        super.setTargetDataSources(targetDataSources);
        super.afterPropertiesSet();
    }

    /**
     * 该方法返回需要使用的DataSource的key值，然后根据这个key从 resolvedDataSources 这个map里取出对应的DataSource，如果找不到，则用默认的 resolvedDefaultDataSource
     * <br/> resolvedDataSources 取值来自 AbstractRoutingDataSource.setTargetDataSources()
     * <br/> resolvedDefaultDataSource 取值来自 AbstractRoutingDataSource.setDefaultTargetDataSource()
     * @return
     */
    @Override
    protected Object determineCurrentLookupKey()
    {
        return DynamicDataSourceContextHolder.getDataSourceType();
    }
}